import torch
class hparams():
    def __init__(self):
        
        # 数据缓存
        self.train_dir_A = "data/S0913/train"
        self.train_dir_B = "data/gaoxiaosong/train"
        self.catch_A = "catch/train_A"
        self.catch_B = "catch/train_B"
        self.n_frames =128
        
        # 特征提取相关
        self.fs = 16000         # 采样率 
        self.frame_period = 5.0 # 帧移
        self.coded_dim = 36     # mepc 特征维度 
        
        # 训练相关参数
        self.g_lr = 2e-4
        self.d_lr = 1e-4
        self.train_steps = 2e4
        
        # learing rate 衰减
        self.start_decay = 1e4
        self.decay_G = self.g_lr/2e5
        self.decay_D = self.d_lr/2e5
        
        # 丢失 identity_loss
        self.step_drop_identity = 1e4
        
        # lamda
        self.identity_loss_lambda = 10
        self.cycle_loss_lambda = 5
        
        #每个 step_log 进行一次保存
        self.step_save = 2000
        
        # 模型保存路径
        self.path_save = 'save3'
        
        
        
         
        
